#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
日志工具模块
提供日志配置和设置功能
"""

import os
import sys
import logging
from typing import Optional

def setup_logger(
    log_level: str = "INFO",
    log_file: Optional[str] = "evaluation.log",
    log_format: str = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
) -> logging.Logger:
    """
    设置日志记录器
    
    Args:
        log_level: 日志级别 (DEBUG, INFO, WARNING, ERROR, CRITICAL)
        log_file: 日志文件路径，None表示不保存到文件
        log_format: 日志格式
    
    Returns:
        logging.Logger: 配置好的根日志记录器
    """
    # 获取根日志记录器
    logger = logging.getLogger()
    
    # 设置日志级别
    level = getattr(logging, log_level.upper(), logging.INFO)
    logger.setLevel(level)
    
    # 清除已有的处理器
    for handler in logger.handlers[:]:
        logger.removeHandler(handler)
    
    # 创建控制台处理器
    console_handler = logging.StreamHandler(sys.stdout)
    console_handler.setFormatter(logging.Formatter(log_format))
    logger.addHandler(console_handler)
    
    # 如果指定了日志文件，创建文件处理器
    if log_file:
        # 确保日志目录存在
        log_dir = os.path.dirname(log_file)
        if log_dir and not os.path.exists(log_dir):
            os.makedirs(log_dir, exist_ok=True)
            
        file_handler = logging.FileHandler(log_file, encoding="utf-8")
        file_handler.setFormatter(logging.Formatter(log_format))
        logger.addHandler(file_handler)
    
    # 输出日志配置信息
    logger.debug(f"日志已配置：级别={log_level}, 文件={log_file}")
    return logger

def get_logger(name):
    """
    获取指定名称的日志器
    
    Args:
        name: 日志器名称，通常为模块名
        
    Returns:
        logging.Logger: 指定名称的日志器
    """
    return logging.getLogger(name)

class LoggerAdapter(logging.LoggerAdapter):
    """
    日志适配器，用于添加上下文信息
    """
    
    def __init__(self, logger, extra=None):
        """
        初始化日志适配器
        
        Args:
            logger: 原始日志器
            extra: 额外信息字典
        """
        super().__init__(logger, extra or {})
    
    def process(self, msg, kwargs):
        """
        处理日志消息，添加上下文信息
        
        Args:
            msg: 原始消息
            kwargs: 关键字参数
            
        Returns:
            tuple: 处理后的消息和关键字参数
        """
        # 如果有用户ID，添加到消息前
        if 'user_id' in self.extra:
            msg = f"[User: {self.extra['user_id']}] {msg}"
        
        # 如果有会话ID，添加到消息前
        if 'session_id' in self.extra:
            msg = f"[Session: {self.extra['session_id']}] {msg}"
        
        return msg, kwargs

def get_adapter(logger, user_id=None, session_id=None):
    """
    获取带有上下文信息的日志适配器
    
    Args:
        logger: 原始日志器
        user_id: 用户ID
        session_id: 会话ID
        
    Returns:
        LoggerAdapter: 配置好的日志适配器
    """
    extra = {}
    if user_id:
        extra['user_id'] = user_id
    if session_id:
        extra['session_id'] = session_id
    
    return LoggerAdapter(logger, extra) 